import turtle as tu
import random as ra
import math

tu.setup(1.0, 1.0)
tu.screensize(1.0, 1.0)
t = tu.Pen()
t.ht()
colors = ['pink']

class Heart():    #每个爱心（爱心类）

    def __init__(self):
        self.r = ra.randint(10,15)        #爱心的半径
        self.x = ra.randint(-1000,1000)   #爱心的横坐标
        self.y = ra.randint(-500,500)     #爱心的纵坐标
        self.f = ra.uniform(-3.14,3.14)   #爱心左右移动呈正弦函数
        self.speed = ra.randint(5,10)     #爱心移动速度
        self.color = ra.choice(colors)    #爱心的颜色
        self.outline = 1                  #爱心的外框大小（可不要）

    def move(self):                       #爱心移动函数
        if self.y <= 500:                 #当爱心还在画布中时
            self.y += self.speed          #设置上下移动速度
            self.x += self.speed * math.sin(self.f)    #设置左右移动速度
            self.f += 0.1                 #可以理解成标志，改变左右移动的方向
        else:                             #当爱心漂出了画布时，重新生成一个爱心
            self.r = ra.randint(10,15)
            self.x = ra.randint(-1000,1000)
            self.y = -500
            self.f = ra.uniform(-3.14,3.14)
            self.speed = ra.randint(5,10)
            self.color = ra.choice(colors)
            self.outline = 1

    def draw(self):       #画爱心函数，就是用turtle画爱心
        t.pensize(self.outline)
        t.penup()
        t.color(self.color)
        t.goto(self.x, self.y)
        t.pendown()
        t.begin_fill()
        t.fillcolor('pink')
        t.setheading(120)
        t.circle(self.r, 195)
        t.fd(self.r * 2.4)
        t.lt(90)
        t.fd(self.r * 2.4)
        t.circle(self.r, 195)
        t.end_fill()
Hearts = []            #用列表保存所有爱心
for i in range(100):
    Hearts.append(Heart())
tu.bgcolor('black')
while True:           #开始漂浮
    tu.tracer(0)
    t.clear()
    for i in range(50):    #在画布中设置50个漂浮的爱心
        Hearts[i].move()
        Hearts[i].draw()
    tu.update()
tu.mainloop()